import Icon from '@/components/Icon'
import Input from '@/components/Input'
import NavBar from '@/components/NavBar'
import { baseURL } from '@/utils/request'
import { getTokenInfo } from '@/utils/storage'
import { Toast } from 'antd-mobile'
import { useEffect, useRef, useState } from 'react'
import { io } from 'socket.io-client'
import styles from './index.module.scss'

export default function ProfileChat() {
  //   定义数据类型
  type Message = {
    //   用于区别消息类型的 type
    type?: 'robot' | 'user'
    msg: string
    timestamp: number
  }
  //  用于受控组件, 收集输入框数据
  const [msg, setMsg] = useState('')

  //   聊天列表
  const [list, setList] = useState<Message[]>([])

  const clientRef = useRef<any>(null)

  //   点击发送按钮
  const handleSend = () => {
    //   空提示
    if (!msg) return Toast.show({ content: '请输入内容' })
    // 处理用户消息结构
    const userMsg: Message = {
      type: 'user',
      msg: msg,
      timestamp: Date.now(),
    }

    // 后端规定：发送与接收消息的事件(event) 均为 message
    // 发送消息用：  .emit('message')
    clientRef.current.emit('message', userMsg)

    // 追加用户信息
    setList([...list, userMsg])
    // 清空输入框
    setMsg('')
  }

  useEffect(() => {
    // 创建 socketio 客户端实例
    const client = io(baseURL, {
      query: {
        token: getTokenInfo().token,
      },
      transports: ['websocket'],
    })
    // 储存到 ref 方便其他地方使用 client 实例
    clientRef.current = client

    //  首次连接 - 可用于做一些数据初始化
    client.on('connect', () => {
      //   console.log('websocket 连接成功...')
      setList([
        {
          type: 'robot',
          msg: '亲爱的用户您好，小智同学为您服务。',
          timestamp: Date.now(),
        },
        {
          type: 'user',
          msg: '你好啊',
          timestamp: Date.now() + 1,
        },
      ])
    })
    // 后端规定：发送与接收消息的事件(event) 均为 message
    // 接收消息用:   .on('message')
    client.on('message', (data) => {
      // 使用箭头函数方式更改值 可以获取到最新数据
      setList((oldList) => {
        return [...oldList, { type: 'robot', ...data }]
      })
    })

    // 清理函数
    return () => {
      // WebSocket 连接一旦建立就不会自动关闭，组件卸载时需要手动关闭连接
      client.close()
    }
  }, [])

  //  当 List 列表发生变化时, 可视区滚动到底部
  useEffect(() => {
    window.scrollTo({
      // 页面有多高 ,就滚动多少
      top: document.body.scrollHeight,
      // behavior: 'smooth', 平滑滚动
      behavior: 'smooth',
    })
  }, [list])

  // 按回车键发送
  const handleKeyUp = (e: React.KeyboardEvent<HTMLInputElement>) => {
    if (e.key === 'Enter') {
      handleSend()
    }
  }

  return (
    <div className={styles.root}>
      {/* 顶部导航栏 */}
      <NavBar>小智同学</NavBar>

      {/* 聊天记录列表 - 开始 */}
      <div className="chat-list">
        {list.map((item) => {
          // 根据 type 区分消息类型
          if (item.type === 'robot') {
            return (
              <div key={item.timestamp} className="chat-item robot">
                <Icon type="iconbtn_xiaozhitongxue" />
                <div className="message">{item.msg}</div>
              </div>
            )
          } else {
            return (
              <div key={item.timestamp} className="chat-item user">
                <img
                  src={'http://toutiao.itheima.net/images/user_head.jpg'}
                  alt=""
                />
                <div className="message">{item.msg}</div>
              </div>
            )
          }
        })}
      </div>
      {/* 聊天记录列表 - 结束 */}

      {/* 底部消息输入框 */}
      <div className="footer">
        <div className="input-footer">
          <Icon type="iconbianji" />
          <Input
            autoFocus
            onKeyUp={(e) => handleKeyUp(e)}
            className="no-border"
            placeholder="请描述您的问题"
            value={msg}
            onChange={(e) => setMsg(e.target.value)}
          />
          <span className="send" onClick={handleSend}>
            发送
          </span>
        </div>
      </div>
    </div>
  )
}
